# -*- coding=utf-8 -*-
import os
import re


def remove_chinese_from_files(input_folder, output_folder):
    # 检查输出文件夹是否存在，如果不存在则创建
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)
    # 遍历输入文件夹中的所有文件
    for root, dirs, files in os.walk(input_folder):
        for file in files:
            # 只处理.txt 和.lrc 文件
            if file.endswith('.txt') or file.endswith('.lrc'):
                input_file_path = os.path.join(root, file)
                print('current file : ',file)
                output_file_path = os.path.join(output_folder, file)
                with open(input_file_path, 'r', encoding='gbk') as input_file:
                    lines = input_file.readlines()
                new_lines = []
                for line in lines:
                    # 使用正则表达式匹配中文字符，如果没有中文字符则添加到新行列表中
                    if not re.search(r'[\u4e00-\u9fff]+', line):
                        new_lines.append(line)
                with open(output_file_path, 'w', encoding='utf-8') as output_file:
                    output_file.writelines(new_lines)

if __name__ == '__main__' :
    # 使用示例
    input_folder = 'C:/KwDownload/temp/'
    output_folder = 'C:/KwDownload/Lyric/'
    remove_chinese_from_files(input_folder, output_folder)